Minor change to main.js