Handle loading in controller