Improve example slightly