fixed UI a bit, added end of game message, fixes #18