Edited __init__.py to import yaml correctly when using dist package