Add more testing code.