updating assigment 5