Main klasse aangepast