Updates to models and db