Updating for challenge 3