Updated the example with the framework