Added input feature. Now it works