cleaning up
small UI changes