Add test runner that actually works