Small js change to add and remove classes when doing expand/collapse all

Fixes #97