Refactored FunSuite tests to use FlatSpec instead