fixing up figures