added saving and more logging to experiment