Create README.md
Refactor code