Add actual working test suite