Fix test and add more comments