changed code

updated code