added another case

fixed some silly mistakes