Updating for assignment 4.