Add Square class