added some additional code for testing purposes