Minor changes

Gave some modifications a try. Didn't work. Ended up just fixing the
script alittle. nothing big.