Add test. Adjust class.