added LSTM visualization.