Use itemgetter instead of lambda expression