Started work on the UI. Model is currently bugged.