homework minor fix