Started work on the main UI