This contains visualization of assignment 2.1