adding remove logic