Added better plots and more detailed solution code