Use Testify library for testing