Updated add diagnosis wizard