code modified

code improved