Reuploading for fixing a minor issue, the previous file had max_states_global = 10 instead than max_states_global = 20 so now the script should work as intended.