Changed the class Box