Making callback optional 

This pull request fixes #9
let me know if you have any comments