Homework update 4