small update to models