Removing unecessary class