Updates to PM allocator