add displaying part within Generator class