Use `take` rather than `mask` in projection

Also fix some of the documentation