update .py
format code