Added tests for ex 1