Setting up Test for multiple classes