Fix toggleClass method