Updated the example slightly