Added Square class