Add testing for new config class