Made slight changes