Wrote quick setup.py
Began rewriting parts to be nicer