Fixed formatting error in python code