week1/assignment2 really done
Fixed bug due to inadequate test cases