Additional testing
Started testing with full dataset