Added save stats

This is buggy and needs to be moved