Added missing class to hide/unhide buttons