UI changes

changed the UI and made button imageButton instead