Add method to refresh size