Small fix to code. No longer need to rely on permute.py and fixed permute function inside of code.