Adding code and plot for the first assignment