added logger to display class
exetended main.py