Changed end scene a bit