Added tests for modified behaviour. Modified README.rst