Laatste fixes voor cleanup

Laatste fixes voor cleanup