first set of tests working