slight changes to models