Prototype code MOSTLY working, but only seems to be one-shot?