Updated RNN test to store the activations as well as LSTM test