Vicky Kaushal back to work