changed the ui a little