Updated test class for Work and Home