change main to a class rather than just w/e