Feature/add unit test

getting better...