refactor

pull class to own file - maintain single main.py file for readability