Use the code from [here](https://github.com/maurergroup/dfttoolkit/blob/79481ab6ba28676c4a431bdb8e9f96cb3037427b/dfttoolkit/cube.py#L432) and integrate with existing cube visualisation module