Add: afterLoop now works