Moved key parameter to __init__