Improving readme, adding more tests