fix python code formatting