Modifying class toggle code