Updated to take in full file, not small dataset