Updated readme, improved some visualization logic