left over? The __init__ below does not seem to have an `axis` argument