Another bug fix in generator
Change version to 0.1.1