Addressed review comments. Change shape of data file. Docs fixes. Added more tests.