Use some more scala magic
Flatten methods