updated docs, fixed test