moving main file to class
made imagine a class
cleanup