De code een beetje opgeruimd en formatting toegepast