Updated with code/ figures for assignment