Add first part of Homework 5