Updated UI and added style