Added __init__ and setup