TODO add agent CLI arguments improvements