Updated for baseline

Updated file to run baseline analysis. Modified code to use dictionary
instead of excessive elif statements.