Added input functionality