Could not get remove to function properly