add support for multiple partitions