Added stats printing to main
Removed iterations from mm inference