Adding test for room generation