A little bit of additional testing code