update code for style