Add states for menu buttons