step 3: add some tests