[IMP] : code as per new api