Week 1 project

Final code and plots