Add traits for mmul_transformer
Cleanup
Use final