MODIFIED - use module.exports correctly