How does atomospy simulate Lorentz Transform?

[uditagarwal97]

via images? animations? models? or what

[Shikhar1998]

For now, I have implemented it using 3-D plots, you can find it here.
This code involves plotting (x, y, z, t) i.e 4 dimensions using 3 space axis's and color gradient feature. In future, I intend to use graphics library of python to implement the same using animation. I believe this would be even a better add-on.