Generating basic plot for hw6
Generating Dk (need to refactor)
Generating 2nd figure.
Was using old variable in function, so broke if run outside my workspace