Updating for homework 5