cater for other courses

1 files changed, 1 insertions, 1 deletions

diff --git a/lib/login.js b/lib/login.js
index c28c334..694b2a5 100644
--- a/lib/login.js
+++ b/lib/login.js
@@ -113,7 +113,7 @@ module.exports = {
                     if (current.text().indexOf('Video') !== -1) { list.push(current.attr('href')); }
                 });
 
-                getCourses = _.filter(list, function map(item) { return (item.indexOf('wiki/M101') !== -1); });
+                getCourses = _.filter(list, function map(item) { return item.match(/(wiki\/M101|wiki\/M102|wiki\/C100|wiki\/M202)/); });
 
                 callback(null, _.map(getCourses, function map(item) { return { name: item.replace(tag, '').slice(0, -1).slice(1).replace(/-/g, ' ').replace(/\//g, ': '), value: item }; }));
             }