Close #140
Allows you to inject the input inside the model