That AI robot that had an emotional meltdown in space got an upgrade