Added tests

Addresses #30
If we are gona change this just add a new line while we are at it haha