Review follow up

- change implementation to use toggleClass()