X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=06-tour-shapes%2Ftour-shapes-build-problem-set.ipynb;h=5d9c57dd6396809f6d02e2f2d4cc83f3f481cb71;hb=a411357b201d8391e5e7c23893fca8705fd4cced;hp=05df3100e5e578478bbb4fa4a1b6cf51670f6d81;hpb=9016ac45d2a303fbd6499a902cce3619d1f4f1d4;p=ou-summer-of-code-2017.git

diff --git a/06-tour-shapes/tour-shapes-build-problem-set.ipynb b/06-tour-shapes/tour-shapes-build-problem-set.ipynb
index 05df310..5d9c57d 100644
--- a/06-tour-shapes/tour-shapes-build-problem-set.ipynb
+++ b/06-tour-shapes/tour-shapes-build-problem-set.ipynb
@@ -278,9 +278,14 @@
    },
    "outputs": [],
    "source": [
-    "def plot_trace(trace, colour='k', xybounds=None, fig=None, subplot_details=None, filename=None):\n",
+    "def plot_trace(trace, colour='k', highlight_start=True,\n",
+    "               xybounds=None, fig=None, subplot_details=None, filename=None):\n",
     "    plt.axis('on')\n",
     "    plt.axes().set_aspect('equal')\n",
+    "        \n",
+    "    if highlight_start:\n",
+    "        plt.axes().add_patch(plt.Circle((trace[0].x, trace[0].y), 0.2, color=colour))\n",
+    "        \n",
     "    for s, t in chunks(trace, 2):\n",
     "        w, h = plot_wh[t.dir]\n",
     "        plt.arrow(s.x, s.y, w, h, head_width=0.1, head_length=0.1, fc=colour, ec=colour, length_includes_head=True)\n",