Updated the UI to match that of the new update currently being worked on for the full game and fixed up small remaining problems that were still in the demo.