Fix for #1

State of the square now matches user's actions rather than what is on screen