Update 531.py

Got rid of unnecessary lines