add results saving and ploting