Changing signature of the initializer closure

Fixes #13 and improves documentation