Add toggle-button class on init