Keras lstm equations, matrices and diagram

I was reading this post, mainly the LSTM equation section for understand the structure diagram and relate it with the matrices and I can deducted this cell diagram:

I made this homemade diagram because in the Colah’s article it’s not the exactly same cell that in keras.

I need know if it ok, can someone confirm me that? This are the equations

W are weights matrices for input data and U are weights matrices for previous hidden state

b are biases

In the pink circles there are operations. The point it’s multiply element by element and X is matrix multiplication