Updating my main page code