cleaned up generation of datasets to allow te user to specify the input dataset rather than have it hardcoded
added new models and model templates