Fixing mainjs file