Ah! I can fix that and adapt the code