updated for local LSTM.py