Added code and modified README.md