Improve module generator