Forgott to add custom Repository Class