Sequential Memory Unit: Building an RNN Cell - Coding problem | OneNoughtOne