minor code update ..