Minor edit to main.js