edit model generator for my needs