Moved weight normalization in transfmat function