Tested fix

Just a quick update. My code been tested, and now works.
Note: I have not tested Doomlaser's