Removed pickle, main returns data