make it a little better

adding some more things to make the script more useful