made some minor edits to run script
Spruced up a few things in main