Add input

Sorry think in another pull request missed something, I think that everything's ok. I hope you enjoy