Updated the example accordingly