Added __init__ file so it can be directly used as a module