Update rnn_features.py