Added the README.md file and minor refactor