updated test and some doc comments.