Updated to reflect the code in run_analysis.R