minor change as per comments