Junit tests added