Updates to HW6. Added more plots and formatting.