Added README.md. Small changes in the code