Made minor changes to two lines