Updating project 2 assignment with my implementation