Added fig/ax arguments in frontogenesis