minor update to models