Add setup.py and rearrange dir