Removed a print statement wiht a hardcoded reshape() command which breaks
the code for arbitrary tau != 3.