added code to test