Fixed code, updated README.md