Added square and rectangle to test inheritance.