Add more test for DataFrame