Fixed the input code