Code beetje opgeruimd.