change to README.md and main python file