updated code

change some code here