Added UI class, fleshed out game logic