minor tweak: set partitions for flights dataset based on input