add draft code for loading screen