Removed square restriction for tril/triu (part 2)