added some code for testing purposes