Fixing top comment in Problem 4