Updated README, added more test cases, fixed imports and changed tabs to spaces