small changes (comments and some asserts added)