Thank you all for the comments and showing the weaknesses in the model and visualisation. I'll try to understand the issues and fix them soon.

I've just published the first batch of patches and new features. I've learnt a lot during the process and from the comments which was one of the main goals, so I'm really happy about this process. Thanks again!