Feature/update repository implementation