Added more main UI functionality.