Feature/add test