Temporary working solution. Still need to modularize and make setup.py