Add: tests
I HATE MYSELF AND I WANT TO DIE