Working on the class box