Finished basic version of UI