fixed bug where we werent merging the arguments after matrix multiplications