Improve arguments handling