Use ndim from PyPI to make it easier to get and run from github.