clean up bugs in example