Fixed #2 and updated README.md