Pooja: made changes to visualization model and controller to reflect current refcatored code