added the option to save the rnn on display