Added Square as type of rectangle
Forgot to import rectangle into square