Update README.md

cleaning up the code structure