Changed lines

Changed code for pull request.