few last changes to the figures