Updated codes

Updated codes