Updated the example a bit