Upgrade CodeGenerator to 0.4.1
Merge pull request #11 from tomaszdurka/issue-11

Upgrade CodeGenerator to 0.4.1