add rest of tests