test 1 and 2 working for part 2