Fixed the ending, which was a bit lame