Figures as well

Final assignment 1