Fix the main.go to work with the new refactoring of the folders