Add test class
Make transfer class public and testable