Adding allocator support
Updated docs