code veranderd na feedback van David