Add tests

Is there an upgrade path for users of the old version?