Added tests. All passing. Fixed README