updated code based one new code