added states to __init__