Add custom rake methods