Adding basic unitest