Update build_data.sh so it can be used with both datasets