Small changes in __init__py