Add changes to NewClass