Couple of tests added