Former England batter Gary Ballance plans to restart his international career with Zimbabwe, after switching his allegiance back to the country of his birth.