added removal functionality