Code verbeterd en opgeruimd