add documentation / fix test