polished up example in code and readme