Upgrade CodeGenerator to 0.4.1