modified square.class