Add callback and modified assert.js