This function was not working if the figure had no subplots. Maybe now it does not work with subplots