Made some slight changes to the first two problems