Pooja: fixed the rspec for visulization