Updated script.js

Added the input section to script.js