homework 5
code