Extracted tests to own source file