Added a call back function with assertions to complete my test case.