Update 531.py

I prefer having imports like this.