Added the py file and updated the readme.