FIX: Removed the hard-coded size for the segment tree