Finished preliminary UI elements - need to work on controller class