Optimize the loading UI