Fixed some comments in the Python code