First version finished for python square and rectangle