Building an LSTM cell