Add test code to README