Finished up the UI, need to make it functional