Start working on version 0.2
Make tests more robust