updated generator to support 0.4.4