Task 4: UI.

- Getting the UI ready for Task 4.
- Some fixes.