Update square.java