made minor changes to models