Adding more functionality to the example.