fixed code, added to readme.md