Add parameter for inject script
Closes #1.