Solution for W2 - Assignment 1