Added some code to test