Adding some testing code