Adding a callback on initialize
Version bump to 2.0.1