Commiting Yogi's solution to Programming Assignment 2