display func is now in its own class