Adding some asserts in example