Couple of test scenarios not accounted for. Added them in.