Adding my code

completed Assignment 2