zram: Cleanup

* This needs to look more like upstream to prevent future merge issues

Signed-off-by: Chester Kener <Cl3Kener@gmail.com>