Add the .flatten() method - all working lovely