Make ButtonState a struct again but add a method `Set` so it doesn't need to be a class for the property