Fix count. Add documentation