Add take as function