Everything is operational!
States now work properly.
Trying to make the buttons bigger