Update rake task

Closes #7