Add tests
fix cs