UI and controller

Added UI changes and created a controller