Documentation figure fixes