Made minor Changes

Modified code slightly