What improvement did von Neumann see in the nuclear weaponry?