Make 2 new classes, ExtendPunch and RetractPunch in the place of Punch