Update of code

Updated Code