Adding missing callback function