Moving game code into Unity