hw-2: added missing import and corrected file type