Added some input functionality