Fixing button toggle example