update code to working