Can now choose between diagnose or medicine