Add models and controller