Toggling class

cleaning the code