added some code for testing