Updating initial code