Add unit tests
IGNORE ALL THE THINGS