Feature/ui

- [x] finish the code
- [x] gather feedback for my changes


| Q             | A
| ------------- | ---
| Bug fix?      | yes
| New feature?  | no
| Tests pass?   | none
| Fixed tickets | none
| Doc           | none