added base code for homework 4