Small improvements on the graphics in methods section