Fixed imports
Added package info to setup.py