Added input button functionality