annotate array shape, faster test, add to example