changed the ui a little bit