Added more code to game