Python code: removing comments