Updatig the code