add unit test and remove unnessary package