Add layout support similar to the hist plot layout

Add an optional layout keyword argument in order to set the subplots layout