Update partition_setup.sh