Added diagnostic model