minor changes to models