Python Heart Shape Code