Added square method
Added last 2 functions