Added a bit of UI