Updating initial code developed