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

Fixes #97
Merge pull request #111 from skiesel/dev/skiesel

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