Week 3 Updated Code with PLOTS