minor fix; callback function added