some minor improvement in problem set4
the bug free things