Improving figures in implementation