remove method working fine now