fixed bugs, implemented fully working viz