Code For Perfect Square In Python