Input code fixes