Add some minor test code