Exercise4: minor correction