BalajiN | Adding back concept class for dagnosis