Remove flatten. Issue with cons