Updated code to account for people cloning the repo and to help it run right out of hte box.