Add repository handling