Small cleanup and updated readme.md with example