Allow setting custom arguments