Extend @return array comments