Added tests

Addresses #30
Naw, that's fine :) Just sounds like a dumb thing to respond with :D