Move state calculation to Button class