Change figure and code