Add a class Room, to made this program more understandable and intelligent