So, you have some code that works, for some definition of work.
You have some other code that does not work, for some other definition of work.
Yet, you did not post that code, or that definition of work. In spite of that, you expect us to tell you what you need to change to make the code work. What you been smoking, man?