EDIT: Edited the visualization functions.