improve format of figure