Update 531.py

cleaned up imports some more