updated the __init__ file