Coding an LSTM