added ending message. need to fix unitsetup