diff --git a/labs/lab05/code/inlab/BinarySearchTree.cpp b/labs/lab05/code/inlab/BinarySearchTree.cpp
index 7d6250abd..dda5b0296 100644
--- a/labs/lab05/code/inlab/BinarySearchTree.cpp
+++ b/labs/lab05/code/inlab/BinarySearchTree.cpp
@@ -41,7 +41,8 @@ BinaryNode* BinarySearchTree::remove(BinaryNode*& n, const string& x) {
     return NULL;
   }
   // first look for x
-  if (x == n->value) {
+  int cmp = n->value.compare(x);
+  if (cmp == 0) {
     // found
     // no children
     if (n->left == NULL && n->right == NULL) {
@@ -68,7 +69,7 @@ BinaryNode* BinarySearchTree::remove(BinaryNode*& n, const string& x) {
     string sr = min(n->right);
     n->value = sr;
     n->right = remove(n->right, sr);
-  } else if (x < n->value) {
+  } else if (cmp < 0) {
     n->left = remove(n->left, x);
   } else {
     n->right = remove(n->right, x);
@@ -117,8 +118,9 @@ void BinarySearchTree::printTree(BinaryNode* root, Trunk* prev, bool isRight) {
 
   if (prev) prev->str = prev_str;
   trunk->str = "   |";
-
+  
   printTree(root->left, trunk, false);
+  delete trunk;
 }
 
-void BinarySearchTree::printTree() { printTree(root, NULL, false); }
\ No newline at end of file
+void BinarySearchTree::printTree() { printTree(root, NULL, false); }
diff --git a/labs/lab05/code/postlab/AVLTree.cpp b/labs/lab05/code/postlab/AVLTree.cpp
index 6ca4f70af..915c8c178 100644
--- a/labs/lab05/code/postlab/AVLTree.cpp
+++ b/labs/lab05/code/postlab/AVLTree.cpp
@@ -59,7 +59,8 @@ AVLNode* AVLTree::remove(AVLNode*& n, const string& x) {
     return NULL;
   }
   // first look for x
-  if (x == n->value) {
+  int cmp = n->value.compare(x);
+  if (cmp == 0) {
     // found
     // no children
     if (n->left == NULL && n->right == NULL) {
@@ -86,7 +87,7 @@ AVLNode* AVLTree::remove(AVLNode*& n, const string& x) {
     string sr = min(n->right);
     n->value = sr;
     n->right = remove(n->right, sr);
-  } else if (x < n->value) {
+  } else if (cmp < 0) {
     n->left = remove(n->left, x);
   } else {
     n->right = remove(n->right, x);
@@ -154,8 +155,9 @@ void AVLTree::printTree(AVLNode* root, Trunk* prev, bool isRight) {
 
   if (prev) prev->str = prev_str;
   trunk->str = "   |";
-
+  
   printTree(root->left, trunk, false);
+  delete trunk;
 }
 
-void AVLTree::printTree() { printTree(root, NULL, false); }
\ No newline at end of file
+void AVLTree::printTree() { printTree(root, NULL, false); }