Using momentum from the AOD track