Fixes #10

Although not a great script, it does what it should.