Fix the setup to include data files properly, include yaml files