Better structure for the two children case