Updated challenge to include conditional logic