small comments on the python code