in the middle of adding visualization functions