Finished up the UI