Add ability to config.
Update tests.