Add responsive design
Change in visualization functions