[modify exercise]
change the input type statement.