Add support for init blocks. Fixes #2