fix code from re-factoring while doing unit tests

fixes #29 

this also does not break the main.py script anymore

the main.py script needs to be moved out of this repo #30