Added shape arguments to docstring in alignment