Enhanced the example