Small changes to models