Fixed the setup.py and moved the code in to src