I believe your classmate received a solution already. Check here for the answer by aaron_baugher whose solution doesn't load everything into memory.