Code and figures for assignment submission