Adjusted code and figure size