Wrote more visualization and added figures