Add some code to keep track of the rooms