Add Tests

Adds tests. Increases test coverage from 30% to 70% :smile_cat: 
Not a big deal, but I have not had good luck with equals-aligned variables in the past so I would suggest only one space before and after the equals sign.