Working on the class