updated example, fix for dim order issue