fixing issue #1 fixed, just forgot to add the modified script, sorry