Add Tests

Adds tests. Increases test coverage from 30% to 70% :smile_cat: 
What should happen here? I don't understand what the test actually tests.