Meer cleanup van de code, poging tot de verbanden goed leggen :p