Code opgeschoond en gesimplificeerd