Updaing the code