Added more robust example of using framework