small fix
minor fix in case B
worked on discussion