Refactored 1st challenge test to have a data provider