adding in class code