Write some tests and configuration class