Add take funcntionality