Improving LSTM test