Finalising visulization, fixed some inconsistencies, almost ready for first real test