Fixed a couple bugs, added imports to __init__.py and updated README