Update 531.py

Fixed re