Removing unwanted classes