Made some slight changes!