Added tests

Addresses #30
Not getting the context. Can you elaborate it?