Adding __init__.py so it is easier to use as a library