Updating example code for version 0.2