Changes to the code again

Guys, I think i fixed it. Sorry to worry everyone