Updated ES to handle more arguments