fixed the missing piece

now it works i think