Updating the visualization logic

Adding the rest of the visualization logic and updating the rendering to
be more user friendly

DB-4
DB-29 #time 2h #resolve