Feature/add basic test