Add header to rnn_model.py and define RnnModel class