Added some code for test