refactored folder structure, added an __init__.py to make the code a proper module