Slight modifications to red-black tree