Adding asserts to test