improve example slightly