Feature/add tests coverage