Adding test for relations class