Fixing main.js file
Set up function to include image icons.