removed class from code