Enkele kleine aanpassingen aan de klassen