Update main_cluster.py