add some changes to code