Update tests & docs