Rename function to be ```plot_2d``` ... will also require associated changes to all function calls i.e. line +114 only.