Added take functionality