Add take function