Added in good and bad end screens