Modify README.md
Add licence header in python file
Add licence header in python file