added comment and fixed one test