Edit homework 5