Minor fixes to fifth homework