Add some flexibility for zram