Add partial classes for room and album