changes to UI

needs improvement and better UX design