Moved setup to __init__ for subclasses