Configure rake tasks to accept arguments