Updated the mainLoop