Finished implementing the UI