add Test Code , bug fix