Added source code for Homework 4