Added obvious controls in the main loop