Add tests for DataFrame