Added ability to pass rake arguments