Cleaning up the example function