Game works, but now going to add functionality/refactoring