Update torch image :art: