import Max added in layers.models