homework 6 solution
coursework_7
fix a minor bug