Update README

Wrote a python code