Initial version of code file that will be used to create the plots for the assignment