Add all stage layouts
New learning dataset