1 files changed, 10 insertions, 11 deletions
diff --git a/Rakefile b/Rakefile
index 839a9c3..56f53f3 100644
--- a/Rakefile
+++ b/Rakefile
@@ -1,23 +1,22 @@
 require 'date'
 require 'rake/packagetask'
 
-INCLUDED_FILES = [
-  'Makefile',
-  'README.pdf',
-  'bin/.keep',
-  'exercises/**/*',
-  'src/**/*',
-  'tmp/.keep',
-].freeze
-
 Rake::PackageTask.new("COMP325-2", DateTime.now.strftime("%y%m").to_i) do |package|
   package.need_tar_gz = true
-  package.package_files.add INCLUDED_FILES
+  package.package_files.add [
+    'README.pdf',
+    'bin/*',
+    'test/**/*',
+    'tmp/.keep',
+    '.gitmodules',
+    '.gitignore',
+  ]
 end
 
 task(:pdf) { sh 'node ./node_modules/.bin/mdpdf README.md' }
 task(:clean) { sh 'rm -fr pkg' }
-task default: [:clean, :pdf, :repackage] do
+task(:test) { sh 'bin/test' }
+task default: [:clean, :test, :pdf, :repackage] do
   Dir['pkg/*.tar.gz'].each do |file|
     FileUtils.mv file, file.gsub('-', '_')
   end