Added toggleClass function