Update main.c

Add red black tree test function.