small change to models