Loading screen logic changed