Updated the example and readme