My solution to Assignment 2 - commit 2