Add plots and code for course project