Removed unecessary itemgetter import