fixed the ui, and made it look better