Add missing fields to models