Add figsize to matplotlib setup