Adding restart, fix for #54

I wasn't sure how to add a unit test.  The existing tests don't pass for me, so there wouldn't be much of a point in adding another one.  Manual testing worked well enough.