adding the code for hw03