Add a bit of code for testing.