Add some new tests to the 2nd FP lab