Add models and update DB schema