chore(ui): use default layout with improvements

[rhodinemma]

Description

• This PR adds UI improvements along with layout.

Type of change

• Bug fix (non-breaking change which fixes an issue)
• New feature (non-breaking change which adds functionality)

Trello Ticket ID

Please add a link to the Trello ticket for the task.

How Can This Been Tested?

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce. Please also list any relevant details for your test scenario/configuration

Checklist:

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have commented my code, particularly in hard-to-understand areas
• My changes generate no new warnings
• Any dependent changes have been merged and published in downstream modules

Screenshots

Add relevant screenshots incase they are relevant