From my point of view, since your implementation didn't use the updated node embeddings as input between epochs, it can be seen as a one layer model, right?
So how to add more layers in your model? Have you ever done some experiments about the performance of different layers?