Add figures for assignment 1