more tests added to cover more code