Added a setup.py
Fixed some imports.