Changed the .yaml file to correspond better to our needs.