Add tests for diff