add some tests & asserts