cleaning some of the ui up