Adding code and plots for Assignment 1.