Python 3 LSTM