Added setup.py
Fixed imports