Button and input handler now has logic to only handle input in certain game states