Sameer Wankhede, who arrested Shah Rukh Khan’s son Aryan Khan, set to return to Mumbai