Change call signature of create_layout to better accommodate previous versions of nx