Added tests for basic use cases