Clean up ModelGenerator and extends it to FileGenerator class