Added some testing code