Add parallel states wrapper