Improved readme and annotation in the python code.