Cleaning up figure numbering script