Add refund method to test class