Updated example code and readme