UI Update

Added more code to UI