adding diagnosis prototype 2