changed figsize method