Added Config to allow users to provide configuration
Trying to make tests more robust