UI improvements, stability improvements

Working on making the game functional!