1 files changed, 6 insertions, 4 deletions

diff --git a/src/main.c b/src/main.c
index f3e455b..8af95d6 100644
--- a/src/main.c
+++ b/src/main.c
@@ -200,9 +200,9 @@ void free_avl_tree_nodes(struct AVLNode* root) {
   }
 }
 
-void increase_label_area(struct AVLNode* root, uint16_t label) {
+bool_t increase_label_area(struct AVLNode* root, uint16_t label) {
   if (root == NULL) {
-    return;
+    return FALSE;
   }
   struct MaskData* data = (struct MaskData*)root->data;
   if (data->label == label) {
@@ -214,11 +214,12 @@ void increase_label_area(struct AVLNode* root, uint16_t label) {
   else if (data->label < label) {
     increase_label_area(root->right, label);
   }
+  return TRUE;
 }
 
-void increase_label_perimeter(struct AVLNode* root, uint16_t label) {
+bool_t increase_label_perimeter(struct AVLNode* root, uint16_t label) {
   if (root == NULL) {
-    return;
+    return FALSE;
   }
   struct MaskData* data = (struct MaskData*)root->data;
   if (data->label == label) {
@@ -230,6 +231,7 @@ void increase_label_perimeter(struct AVLNode* root, uint16_t label) {
   else if (data->label < label) {
     increase_label_perimeter(root->right, label);
   }
+  return TRUE;
 }
 
 int main(int argc, char** argv)