update TS models, minor changes for fit