Added treat setup