Improve generate code.