split off test into seperate class