Worked the testing code in.