added option to show summary after training