Fixed matrix sizes (removed a .transpose(), unsure if this fixes
everything)