Add machine count flag & wait for all machines to join