Improve example code