working on unittest