fixed case A
small fix
worked on case B
small fix