Improved button menu functionality by refactoring into a controller that saves button state in