Quick fix for homework1