code changes for correct display