Add first set of tests