Changing a few lines.

Added two lines and deleted one.