Some shitty code for visualization - doesn't work, but helped in finding some problems