Updated __init__ and readme