Feature/add code coverage