Removing unneccessary class