Fix bug 3,4
fixbug 3,4