Began editing / adding in Daniel's code to main python script.