Update example code for 0.2