Should properly toggle class now on open/close buttons