Fixing __init__