fixed a bit in the ending