added a bit of UI