remove asserts
added more tests