added asserts to test