fixed basic visualization for the first stage of the thesis + cleaned
and fixed some bugs.