Moving unit tests to separate class.