Added options for which parts of the workflow to run
Cleaning up imports