Adding Square class