-Added new features for remove button