Use custom repository in main.go