changed part of code