Update main.js

input apply code bug